from backend.experiment.args import model_type

if 'pangu' in model_type:
    from backend.experiment.pangu import tokenizer, vocabulary, id2word
elif 'us' in model_type:
    from backend.experiment.us import tokenizer, vocabulary, id2word
else:
    raise ValueError(f'model_type: {model_type}')
